#ifndef AI_AGENT_H
#define AI_AGENT_H

#include <wx/string.h>
#include <vector>
#include <memory>

// 聊天消息结构
struct ChatMessage {
    wxString role;  // "user", "assistant", "system"
    wxString content;
    wxString timestamp;
};

// 基础AI智能体类
class AIAgent {
public:
    virtual ~AIAgent() = default;
    virtual wxString ProcessMessage(const wxString& message) = 0;
    virtual wxString GetName() const = 0;
    virtual wxString GetDescription() const = 0;

protected:
    std::vector<ChatMessage> conversation_history;
};

#endif // AI_AGENT_H
